package com.dexter.year2023.charpter6_tree_level_travel.level2.topic2_2处理每层元素;

import com.dexter.year2023.charpter6_tree_level_travel.level1.TreeNode;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * 515. 在每个树行中找最大值
 * https://leetcode.cn/problems/find-largest-value-in-each-tree-row/
 * <p>
 * 给定一棵二叉树的根节点 root ，请找出该二叉树中每一层的最大值。
 * <p>
 * 示例1：
 * 输入: root = [1,3,2,5,3,null,9]
 * 输出: [1,3,9]
 * 示例2：
 * 输入: root = [1,2,3]
 * 输出: [1,3]
 * <p>
 * 提示：
 * 二叉树的节点个数的范围是 [0,104]
 * -231 <= Node.val <= 231 - 1
 *
 * @author Dexter
 */
public class LC515LargestValues {
    public static void main(String[] args) {
        TreeNode node = new TreeNode(3);
        node.left = new TreeNode(9);
        node.right = new TreeNode(20);
        node.right.left = new TreeNode(15);
        node.right.right = new TreeNode(7);
        // 提交结果：已通过 - 时间击败88.58% 的Java用户，内存击败14.20% 的Java用户
        System.out.println(largestValues(node));
    }

    public static List<Integer> largestValues(TreeNode root) {
        // 结果，存放每层的最大值 - 数组集合
        List<Integer> res = new ArrayList<>();
        // 判空
        if (root == null) {
            return res;
        }
        // 队列 - 链表集合
        LinkedList<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            // 每层最大值
            int levelMaxValue = Integer.MIN_VALUE;
            for (int i = 0; i < size; i++) {
                // max 与 当前节点 比较
                TreeNode t = queue.remove();
                levelMaxValue = Math.max(levelMaxValue, t.val);

                if (t.left != null) {
                    queue.add(t.left);
                }
                if (t.right != null) {
                    queue.add(t.right);
                }
            }
            // 本层遍历结束，存入最大值
            res.add(levelMaxValue);
        }
        return res;
    }
}
